import json
from pyecharts.charts import Map
from pyecharts.options import *

f = open("D:/疫情.txt", "r", encoding="UTF-8")
data = f.read()
f.close()

data_dict = json.loads(data)

cities_data_list = data_dict["areaTree"][0]["children"][3]["children"]

data_list = []
for cities_data in cities_data_list:
    cities_name = cities_data["name"] + "市"
    cities_confirm = cities_data["total"]["confirm"]
    data_list.append((cities_name, cities_confirm))

# print(data_list)
data_list.append(("济源市", 5))

city_map = Map()
city_map.add("河南省确诊人数", data_list, "河南")

city_map.set_global_opts(
    title_opts=TitleOpts(title="河南省疫情地图"),
    visualmap_opts=VisualMapOpts(
        is_show=True,
        is_piecewise=True,
        pieces=[
            {"min": 1, "max": 99, "lable": "1~99人", "color": "#CCFFFF"},
            {"min": 100, "max": 999, "lable": "100~9999人", "color": "#FFFF99"},
            {"min": 1000, "max": 4999, "lable": "1000~4999人", "color": "#FF9966"},
            {"min": 5000, "max": 9999, "lable": "5000~99999人", "color": "#FF6666"},
            {"min": 10000, "max": 99999, "lable": "10000~99999人", "color": "#CC3333"},
            {"min": 100000, "lable": "100000+", "color": "#990033"}
        ]
    )
)

city_map.render("河南省疫情地图.html")
